//  SPEA2_main.java
//
//  Author:
//       Antonio J. Nebro <antonio@lcc.uma.es>
//       Juan J. Durillo <durillo@lcc.uma.es>
//
//  Copyright (c) 2011 Antonio J. Nebro, Juan J. Durillo
//
//  This program is free software: you can redistribute it and/or modify
//  it under the terms of the GNU Lesser General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
//
//  This program is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU Lesser General Public License for more details.
// 
//  You should have received a copy of the GNU Lesser General Public License
//  along with this program.  If not, see <http://www.gnu.org/licenses/>.

package jmetal.metaheuristics.spea2;

import jmetal.core.Algorithm;
import jmetal.core.Operator;
import jmetal.core.Problem;
import jmetal.core.SolutionSet;
import jmetal.operators.crossover.CrossoverFactory;
import jmetal.operators.mutation.MutationFactory;
import jmetal.operators.selection.SelectionFactory;
import jmetal.problems.ProblemFactory;
import jmetal.qualityIndicator.QualityIndicator;
import jmetal.util.Configuration;
import jmetal.util.JMException;

import java.io.IOException;
import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationTargetException;
import java.util.HashMap;
import java.util.logging.FileHandler;
import java.util.logging.Logger;

/**
 * Class for configuring and running the SPEA2 algorithm
 */
public class SPEA2_main {
  public static Logger      logger_ ;      // Logger object
  public static FileHandler fileHandler_ ; // FileHandler object
  
  /**
   * @param args Command line arguments. The first (optional) argument specifies 
   *             the problem to solve.
   * @throws JMException 
   * @throws IOException 
 * @throws NoSuchMethodException 
   * @throws SecurityException 
   * Usage: three options
   *      - jmetal.metaheuristics.mocell.MOCell_main
   *      - jmetal.metaheuristics.mocell.MOCell_main problemName
   *      - jmetal.metaheuristics.mocell.MOCell_main problemName ParetoFrontFile
 * @throws InvocationTargetException 
 * @throws IllegalArgumentException 
 * @throws IllegalAccessException 
 * @throws InstantiationException 
   */
  public static void main(String [] args) throws JMException, IOException, ClassNotFoundException, NoSuchMethodException, SecurityException, InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException {
    Problem   problem   ;         // The problem to solve
    Algorithm algorithm ;         // The algorithm to use
    Operator  crossover ;         // Crossover operator
    Operator  mutation  ;         // Mutation operator
    Operator  selection ;         // Selection operator
        
    QualityIndicator indicators ; // Object to get quality indicators

    HashMap  parameters ; // Operator parameters
    
    indicators = null ;
    String outDirect = "" ;
    String pfDirect = "";
    
    String[] probName = {"jmetal.problems.DTLZ.DTLZ1",
    		"jmetal.problems.DTLZ.DTLZ2",
    		"jmetal.problems.DTLZ.DTLZ3",
    		"jmetal.problems.DTLZ.DTLZ4",
    		"jmetal.problems.DTLZ.DTLZ5",
    		"jmetal.problems.DTLZ.DTLZ6",
    		"jmetal.problems.DTLZ.DTLZ7"};

    for(int probNum = 0;probNum<probName.length;probNum++)
	{
		if (args.length == 1) {
			Object[] params = { "Real" };
			problem = (new ProblemFactory()).getProblem(args[0], params);
		} // if
		else if (args.length == 2) {
			Object[] params = { "Real" };
			problem = (new ProblemFactory()).getProblem(args[0], params);
			indicators = new QualityIndicator(problem, args[1]);
		} // if
		else { // Default problem
				// problem = new Kursawe("Real", 3);
				// problem = new Water("Real");
				// problem = new ZDT1("ArrayReal", 1000);
				// problem = new ZDT4("BinaryReal");
				// problem = new WFG1("Real");

			Class<?> dtlz = Class.forName(probName[probNum]);
			Constructor<?> constr = dtlz.getConstructor(String.class);
			problem = (Problem) constr.newInstance("Real");
			pfDirect = "PFStar/" + problem.getName() + "_M" + problem.getNumberOfObjectives() + ".dat";
			indicators = new QualityIndicator(problem, pfDirect);
			outDirect = "results/SPEA2/" + problem.getName() + "/";

			// problem = new OKA2("Real") ;
		} // else

		algorithm = new SPEA2(problem);

		// Algorithm parameters
		algorithm.setInputParameter("populationSize", 120);
		algorithm.setInputParameter("archiveSize", 120);
		algorithm.setInputParameter("maxEvaluations", 120 * 1000);

		// Mutation and Crossover for Real codification
		parameters = new HashMap();
		parameters.put("probability", 1.0);
		parameters.put("distributionIndex", 30.0);
		crossover = CrossoverFactory.getCrossoverOperator("SBXCrossover", parameters);

		parameters = new HashMap();
		parameters.put("probability", 1.0 / problem.getNumberOfVariables());
		parameters.put("distributionIndex", 20.0);
		mutation = MutationFactory.getMutationOperator("PolynomialMutation", parameters);

		// Selection operator
		parameters = null;
		selection = SelectionFactory.getSelectionOperator("BinaryTournament", parameters);

		// Add the operators to the algorithm
		algorithm.addOperator("crossover", crossover);
		algorithm.addOperator("mutation", mutation);
		algorithm.addOperator("selection", selection);

		int run = 20;
		for (int i = 0; i < run; i++) {
			// Execute the algorithm
			long initTime = System.currentTimeMillis();
			SolutionSet population = algorithm.execute();
			long estimatedTime = System.currentTimeMillis() - initTime;

			// Logger object and file to store log messages
			logger_ = Configuration.logger_;
			fileHandler_ = new FileHandler(outDirect + problem.getNumberOfObjectives() + "D_" + i + ".log");
			logger_.addHandler(fileHandler_);

			// Result messages
			logger_.info("Total execution time: " + estimatedTime + "ms");
			logger_.info("Objectives values have been writen to file FUN");
			population.printObjectivesToFile(outDirect + "FUN_" + problem.getNumberOfObjectives() + "D_" + i);
			logger_.info("Variables values have been writen to file VAR");
			population.printVariablesToFile(outDirect + "VAR_" + problem.getNumberOfObjectives() + "D_" + i);

			if (indicators != null) {
				logger_.info("Quality indicators");
				logger_.info("Hypervolume: " + indicators.getHypervolume(population));
				logger_.info("GD         : " + indicators.getGD(population));
				logger_.info("IGD        : " + indicators.getIGD(population));
				logger_.info("Spread     : " + indicators.getSpread(population));
				logger_.info("Epsilon    : " + indicators.getEpsilon(population));
			} // if
		}
	}
  }//main
} // SPEA2_main.java
